import json
import os

import frappe


def create_ksa_vat_setting(company):
    """On creation of first company. Creates KSA VAT Setting"""

    company = frappe.get_doc('Company', company)

    file_path = os.path.join(os.path.dirname(__file__), '..', 'data', 'ksa_vat_settings.json')
    with open(file_path, 'r') as json_file:
        account_data = json.load(json_file)

    # Creating KSA VAT Setting
    ksa_vat_setting = frappe.get_doc({
        'doctype': 'KSA VAT Setting',
        'company': company.name
    })

    for data in account_data:
        if data['type'] == 'Sales Account':
            for row in data['accounts']:
                item_tax_template = row['item_tax_template']
                account = row['account']
                ksa_vat_setting.append('ksa_vat_sales_accounts', {
                    'title': row['title'],
                    'item_tax_template': f'{item_tax_template} - {company.abbr}',
                    'account': f'{account} - {company.abbr}'
                })

        elif data['type'] == 'Purchase Account':
            for row in data['accounts']:
                item_tax_template = row['item_tax_template']
                account = row['account']
                ksa_vat_setting.append('ksa_vat_purchase_accounts', {
                    'title': row['title'],
                    'item_tax_template': f'{item_tax_template} - {company.abbr}',
                    'account': f'{account} - {company.abbr}'
                })

    ksa_vat_setting.save()
